package com.qianfeng.smsplatform.strategy.filters;

//
//                            _ooOoo_  
//                           o8888888o  
//                           88" . "88  
//                           (| -_- |)  
//                            O\ = /O  
//                        ____/`---'\____  
//                      .   ' \\| |// `.  
//                       / \\||| : |||// \  
//                     / _||||| -:- |||||- \  
//                       | | \\\ - /// | |  
//                     | \_| ''\---/'' | |  
//                      \ .-\__ `-` ___/-. /  
//                   ___`. .' /--.--\ `. . __  
//                ."" '< `.___\_<|>_/___.' >'"".  
//               | | : `- \`.;`\ _ /`;.`/ - ` : | |  
//                 \ \ `-. \_ __\ /__ _/ .-` / /  
//         ======`-.____`-.___\_____/___.-`____.-'======  
//                            `=---='  
//  
//         .............................................  
//                  佛祖镇楼                  BUG辟易  
//          佛曰:  
//                  写字楼里写字间，写字间里程序员；  
//                  程序人员写程序，又拿程序换酒钱。  
//                  酒醒只在网上坐，酒醉还来网下眠；  
//                  酒醉酒醒日复日，网上网下年复年。  
//                  但愿老死电脑间，不愿鞠躬老板前；  
//                  奔驰宝马贵者趣，公交自行程序员。  
//                  别人笑我忒疯癫，我笑自己命太贱；  
//  


import com.qianfeng.smsplatform.common.constants.CacheConstants;
import com.qianfeng.smsplatform.common.constants.InterfaceExceptionDict;
import com.qianfeng.smsplatform.common.model.Standard_Submit;
import com.qianfeng.smsplatform.strategy.apis.CacheService;
import com.qianfeng.smsplatform.strategy.cache.LimitCache;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.Map;
import java.util.Set;

/**
 * Created by Jackiechan on 2022/3/25/下午9:10
 *
 * @author Jackiechan
 * @version 1.0
 * @since 1.0
 */
@Component
public class LimitFilter implements FilterChain{

    private CacheService cacheService;

    @Autowired
    public void setCacheService(CacheService cacheService) {
        this.cacheService = cacheService;
    }

    private LimitCache limitCache;

    @Autowired
    public void setLimitCache(LimitCache limitCache) {
        this.limitCache = limitCache;
    }

    @Override
    public boolean processSms(Standard_Submit standard_submit) {
        Set<Map> allLimits = limitCache.getAllLimits();//获取到所有的限流数据
        long max = System.currentTimeMillis();//当前的时间
        for (Map limit : allLimits) {
            //遍历每一个限流的规则
            double time = (double) limit.get("score");// 限流的时间长度 单位是秒
            String valueString = (String) limit.get("value"); //当前的限流的值是多少
            int limitCount = Integer.parseInt(valueString);//转成数字
            //我们以每次发送的时间作为分数，以发送的时间作为value 保存到zet中
            double min = max - time * 1000;//起点时间

            Long count = cacheService.getSizeByScore(CacheConstants.CACHE_PREFIX_SMS_LIMIT + standard_submit.getClientID() + standard_submit.getDestMobile(), min, max);

                if (count >= limitCount) {
                //当前已经达到限流阈值，应该拦截
                System.err.println("超出限制了");
                standard_submit.setDescription("超出限制次数了,当前限制的策略是"+time+"秒最多发送"+limitCount+"条数据");
                standard_submit.setErrorCode(InterfaceExceptionDict.RETURN_STATUS_LIMIT_ERROR+"");
                return false;
            }
        }
        //保存数据到redis中
        //放在这里会导致这边限流添加了，但是计费拦截了，导致明明没有发送却仍旧被限流，计费放在这个前面也会出现扣了钱但是被限流限制了
        //我们应该拆开，限流的计算放在前面，添加放到后面
        //cacheService.add2Zset(CacheConstants.CACHE_PREFIX_SMS_LIMIT + standard_submit.getClientID() + standard_submit.getDestMobile(), max + "", max);
        return true;
    }
}
